package org.jeecg.system.web.controller;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.dubbo.config.annotation.Reference;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.aspect.annotation.AutoLog;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.system.api.entity.SysDepartRole;
import org.jeecg.system.api.entity.SysDepartRolePermission;
import org.jeecg.system.api.entity.SysDepartRoleUser;
import org.jeecg.system.api.entity.SysPermissionDataRule;
import org.jeecg.system.api.service.*;
import org.jeecg.system.mapper.query.QueryGenerator;
import org.jeecg.system.web.controller.base.JeecgController;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
* @Description: 部门角色
* @Author: jeecg-boot
* @Date:   2020-02-12
* @Version: V1.0
*/
@Slf4j
@Api(tags="部门角色")
@RestController
@RequestMapping("/sys/sysDepartRole")
public class SysDepartRoleController extends JeecgController<SysDepartRole, ISysDepartRoleService> {

	@Autowired
   private ISysDepartRoleService sysDepartRoleService;

	@Autowired
   private ISysDepartRoleUserService departRoleUserService;

	@Autowired
   private ISysDepartPermissionService sysDepartPermissionService;

	@Autowired
	private ISysDepartRolePermissionService sysDepartRolePermissionService;

	@Autowired
	private ISysDepartService sysDepartService;

   /**
	* 分页列表查询
	*
	* @param sysDepartRole
	* @param pageNo
	* @param pageSize
	* @param req
	* @return
	*/
   @AutoLog(value = "部门角色-分页列表查询")
   @ApiOperation(value="部门角色-分页列表查询", notes="部门角色-分页列表查询")
   @GetMapping(value = "/list")
   public Result<?> queryPageList(SysDepartRole sysDepartRole,
								  @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
								  @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
								  @RequestParam(name="deptId",required=false) String deptId,
								  HttpServletRequest req) {
	   QueryWrapper<SysDepartRole> queryWrapper = QueryGenerator.initQueryWrapper(sysDepartRole, req.getParameterMap());
	   Page<SysDepartRole> page = new Page<SysDepartRole>(pageNo, pageSize);
	   LoginUser user = (LoginUser) SecurityUtils.getSubject().getPrincipal();
	   List<String> deptIds = null;
	   if(oConvertUtils.isEmpty(deptId)){
		   if(oConvertUtils.isNotEmpty(user.getIdentity()) && user.getIdentity() == CommonConstant.USER_IDENTITY_2 ){
			   deptIds = sysDepartService.getMySubDepIdsByDepId(user.getDepartIds());
		   }else{
			   return Result.ok(null);
		   }
	   }else{
		   deptIds = sysDepartService.getSubDepIdsByDepId(deptId);
	   }
	   queryWrapper.in("depart_id",deptIds);
	   IPage<SysDepartRole> pageList = sysDepartRoleService.page(page, queryWrapper);
	   return Result.ok(pageList);
   }

   /**
	* 添加
	*
	* @param sysDepartRole
	* @return
	*/
   @AutoLog(value = "部门角色-添加")
   @ApiOperation(value="部门角色-添加", notes="部门角色-添加")
   @PostMapping(value = "/add")
   public Result<?> add(@RequestBody SysDepartRole sysDepartRole) {
	   sysDepartRoleService.save(sysDepartRole);
	   return Result.ok("添加成功！");
   }

   /**
	* 编辑
	*
	* @param sysDepartRole
	* @return
	*/
   @AutoLog(value = "部门角色-编辑")
   @ApiOperation(value="部门角色-编辑", notes="部门角色-编辑")
   @PutMapping(value = "/edit")
   public Result<?> edit(@RequestBody SysDepartRole sysDepartRole) {
	   sysDepartRoleService.updateById(sysDepartRole);
	   return Result.ok("编辑成功!");
   }

   /**
	* 通过id删除
	*
	* @param id
	* @return
	*/
   @AutoLog(value = "部门角色-通过id删除")
   @ApiOperation(value="部门角色-通过id删除", notes="部门角色-通过id删除")
   @DeleteMapping(value = "/delete")
   public Result<?> delete(@RequestParam(name="id",required=true) String id) {
	   sysDepartRoleService.removeById(id);
	   return Result.ok("删除成功!");
   }

   /**
	* 批量删除
	*
	* @param ids
	* @return
	*/
   @AutoLog(value = "部门角色-批量删除")
   @ApiOperation(value="部门角色-批量删除", notes="部门角色-批量删除")
   @DeleteMapping(value = "/deleteBatch")
   public Result<?> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
	   this.sysDepartRoleService.removeByIds(Arrays.asList(ids.split(",")));
	   return Result.ok("批量删除成功！");
   }

   /**
	* 通过id查询
	*
	* @param id
	* @return
	*/
   @AutoLog(value = "部门角色-通过id查询")
   @ApiOperation(value="部门角色-通过id查询", notes="部门角色-通过id查询")
   @GetMapping(value = "/queryById")
   public Result<?> queryById(@RequestParam(name="id",required=true) String id) {
	   SysDepartRole sysDepartRole = sysDepartRoleService.getById(id);
	   return Result.ok(sysDepartRole);
   }

	/**
	 * 获取部门下角色
	 * @param departId
	 * @return
	 */
   @RequestMapping(value = "/getDeptRoleList", method = RequestMethod.GET)
   public Result<List<SysDepartRole>> getDeptRoleList(@RequestParam(value = "departId") String departId){
	   Result<List<SysDepartRole>> result = new Result<>();
	   List<SysDepartRole> deptRoleList = sysDepartRoleService.list(new QueryWrapper<SysDepartRole>().eq("depart_id",departId));
	   result.setSuccess(true);
	   result.setResult(deptRoleList);
	   return result;
   }

	/**
	 * 设置
	 * @param json
	 * @return
	 */
	@RequestMapping(value = "/deptRoleUserAdd", method = RequestMethod.POST)
	public Result<?> deptRoleAdd(@RequestBody JSONObject json) {
		String newRoleId = json.getString("newRoleId");
		String oldRoleId = json.getString("oldRoleId");
		String userId = json.getString("userId");
		departRoleUserService.deptRoleUserAdd(userId,newRoleId,oldRoleId);
		return Result.ok("添加成功！");
	}

	/**
	 * 根据用户id获取已设置部门角色
	 * @param userId
	 * @return
	 */
	@RequestMapping(value = "/getDeptRoleByUserId", method = RequestMethod.GET)
	public Result<List<SysDepartRoleUser>> getDeptRoleByUserId(@RequestParam(value = "userId") String userId){
		Result<List<SysDepartRoleUser>> result = new Result<>();
		List<SysDepartRoleUser> roleUserList = departRoleUserService.list(new QueryWrapper<SysDepartRoleUser>().eq("user_id",userId));
		result.setSuccess(true);
		result.setResult(roleUserList);
		return result;
	}

	/**
	 * 查询数据规则数据
	 */
	@GetMapping(value = "/datarule/{permissionId}/{departId}/{roleId}")
	public Result<?> loadDatarule(@PathVariable("permissionId") String permissionId, @PathVariable("departId") String departId, @PathVariable("roleId") String roleId) {
	   //查询已授权的部门规则
		List<SysPermissionDataRule> list = sysDepartPermissionService.getPermRuleListByDeptIdAndPermId(departId,permissionId);
		if(list==null || list.size()==0) {
			return Result.error("未找到权限配置信息");
		}else {
			Map<String,Object> map = new HashMap<>();
			map.put("datarule", list);
			LambdaQueryWrapper<SysDepartRolePermission> query = new LambdaQueryWrapper<SysDepartRolePermission>()
					.eq(SysDepartRolePermission::getPermissionId, permissionId)
					.eq(SysDepartRolePermission::getRoleId,roleId);
			SysDepartRolePermission sysRolePermission = sysDepartRolePermissionService.getOne(query);
			if(sysRolePermission==null) {
				//return Result.error("未找到角色菜单配置信息");
			}else {
				String drChecked = sysRolePermission.getDataRuleIds();
				if(oConvertUtils.isNotEmpty(drChecked)) {
					map.put("drChecked", drChecked.endsWith(",")?drChecked.substring(0, drChecked.length()-1):drChecked);
				}
			}
			return Result.ok(map);
			//TODO 以后按钮权限的查询也走这个请求 无非在map中多加两个key
		}
	}

	/**
	 * 保存数据规则至角色菜单关联表
	 */
	@PostMapping(value = "/datarule")
	public Result<?> saveDatarule(@RequestBody JSONObject jsonObject) {
		try {
			String permissionId = jsonObject.getString("permissionId");
			String roleId = jsonObject.getString("roleId");
			String dataRuleIds = jsonObject.getString("dataRuleIds");
			log.info("保存数据规则>>"+"菜单ID:"+permissionId+"角色ID:"+ roleId+"数据权限ID:"+dataRuleIds);
			LambdaQueryWrapper<SysDepartRolePermission> query = new LambdaQueryWrapper<SysDepartRolePermission>()
					.eq(SysDepartRolePermission::getPermissionId, permissionId)
					.eq(SysDepartRolePermission::getRoleId,roleId);
			SysDepartRolePermission sysRolePermission = sysDepartRolePermissionService.getOne(query);
			if(sysRolePermission==null) {
				return Result.error("请先保存角色菜单权限!");
			}else {
				sysRolePermission.setDataRuleIds(dataRuleIds);
				this.sysDepartRolePermissionService.updateById(sysRolePermission);
			}
		} catch (Exception e) {
			log.error("SysRoleController.saveDatarule()发生异常：" + e.getMessage(),e);
			return Result.error("保存失败");
		}
		return Result.ok("保存成功!");
	}

// /**
//  * 导出excel
//  *
//  * @param request
//  * @param sysDepartRole
//  */
// @RequestMapping(value = "/exportXls")
// public ModelAndView exportXls(HttpServletRequest request, SysDepartRole sysDepartRole) {
//	 return super.exportXls(request, sysDepartRole, SysDepartRole.class, "部门角色");
// }
//
// /**
//  * 通过excel导入数据
//  *
//  * @param request
//  * @param response
//  * @return
//  */
// @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
// public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
//	 return super.importExcel(request, response, SysDepartRole.class);
// }

}
